package com.spider.silence.worker;

import com.alibaba.fastjson.JSON;
import com.alibaba.fastjson.JSONObject;
import com.alibaba.fastjson.serializer.SerializerFeature;
import com.spider.silence.downloader.ZuiChengduDownLoader;
import com.spider.silence.itf.output.Writer;
import com.spider.silence.output.*;
import com.spider.silence.pool.ZuiChengduDownLoaderPool;
import com.spider.silence.task.ZuiChengDuTask;
import com.spider.silence.task.manager.CommonTaskManager;
import com.spider.silence.utils.DateUtils;
import org.apache.log4j.Logger;

import java.io.File;
import java.util.Map;

/**
 * Created by 夏茂轩 on 2016/12/26.
 */
public class ZuiChengduWorker implements Runnable {
    private Logger logger = Logger.getLogger(ZuiChengduWorker.class);

    @Override
    public void run() {
        while (true && !Thread.interrupted()) {
            ZuiChengduDownLoaderPool pool = ZuiChengduDownLoaderPool.getInstance();
            ZuiChengduDownLoader downLoader = null;
            FileWriter writer = null;
            FileWriterPool writerPool =FileWriterPool.getInstance();
            try {
                downLoader=pool.fetchDownLoader(1000);
                if (downLoader != null) {
                    ZuiChengDuTask task = (ZuiChengDuTask) CommonTaskManager.takeTask("ZuiChengDu",1000);
                    if(task==null){
                        continue;
                    }
                    JSONObject content = downLoader.crawlItem(task.getForumUrl());
                    String filePrefix = "ZuiChengdu";//文件前缀
                    JSONObject data = content.getJSONObject("data");
                    long tsp = DateUtils.dateToTimestamp(DateUtils.parseDateTime(data.getString("publish_time"))).getTime();
                    String user_id = data.getString("publish_user_id");
                    String tid = data.getString("id");
                    String fileName = filePrefix + "_" + tsp + "_" + user_id + "_" + tid;//平台名称（或者英文）_ 言论发布的时间戳 _ 发布用户的ID _ 言论的tid
                    String publishDate = DateUtils.formatDate(DateUtils.parseDate(data.getString("publish_time"))).replace("-", "-");
                    writer = writerPool.fetchWriter();
                    Map<String,File> map=downLoader.getFiles();
                    writer.write("最成都_speeches_" + publishDate, fileName, JSON.toJSONString(content, SerializerFeature.BrowserCompatible),map);
                }
            } catch (InterruptedException e) {
                logger.error(e.getMessage(), e);
            } catch (Exception e) {
                logger.error(e.getMessage(), e);
                e.printStackTrace();
            } finally {
                if(downLoader!=null){
                    pool.releaseDownLoader(downLoader);
                }
                if(writer!=null){
                    writerPool.releaseWriter(writer);
                }
            }
        }
    }
}
